package org.example.binaryTree;

/**
 * 求当前节点的后继节点
 * 1，如果当前节点有右子树，那么右子树的最左节点必是后继结点
 * 2，当前节点无右树，找当前节点是父亲节点的左孩子，没有 null
 */
public class SuccessorNode {
    public static Node getSuccessorNode(Node node) {
        if (node == null) {
            return null;
        }
        //有右子树，则右子树的最左节点为后继节点
        if (node.right != null) {
            return getLeftMost(node.right);
        } else {
            Node parent = node.parent;
            while (parent != null && parent.left != node) {
                node = parent;
                parent = node.parent;
            }
            return node;
        }

    }

    private static Node getLeftMost(Node node) {
        if (node == null) {
            return null;
        }
        while (node.left != null) {
            node = node.left;
        }
        return node;
    }
}
